package lc.com.lcpicturebackend.manager.app.rag;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;
import org.springframework.stereotype.Component;

/**
 * 功能：自定义错误空知识库逻辑
 * 作者：lc
 * 日期：2025/6/4 22:39
 */
@Component
public class AppContextualQueryAugmenter {
    public static ContextualQueryAugmenter create() {
        // 定义空模板
        PromptTemplate emptyContextPromptTemplate = new PromptTemplate("""
                 当上下文为空时你只能输出下面的内容：
                                抱歉，您的问题不在我的回答范围之内，没办法帮到您哦，
                                您可以联系站长 2549615161@qq.com
                """);
        // 返回自定义错误空知识库逻辑
        return ContextualQueryAugmenter.builder()
                .allowEmptyContext(false)
                .emptyContextPromptTemplate(emptyContextPromptTemplate)
                .build();
    }
}
